# test02_unittest.py
# 定义一个函数add，传递任意两个数字，返回总和
# 编写测试类TestAdd，对函数进行测试，每个测试方法中调用函数add
# 最后批量执行测试用例
import unittest
def add(a, b):
    return a + b
# 单元测试：测试函数
class TestAdd(unittest.TestCase):
    def test_add01(self):
        result = add(1, 2)
        print('测试结果：', result)
    def test_add02(self):
        result = add(3, 4)
        print('测试结果：', result)
    def test_add03(self):
        result = add(5, 6)
        print('测试结果：', result)
if __name__ == '__main__':
    # 执行测试类中所有的测试方法
    unittest.main()